python avoid nested for loops